# 流感诊断机器人
import tkinter as tk

def send():
    # 风寒感冒症状
    flu1 = {'发热': '0', '寒气重': '1', '头痛': '1', '鼻塞': '1', '流黄鼻涕': '0', '流清鼻涕': '1'}
    # 风热感冒症状
    flu2 = {'发热': '1', '寒气重': '0', '头痛': '0', '鼻塞': '1', '流黄鼻涕': '1', '流清鼻涕': '0'}

    # 风寒计算
    sum1 = 0  # 命中数
    for k in flu1:
        if checkboxes[k].get():
            ok = 1
        else:
            ok = 0
        if ok == flu1[k]:
            # 命中
            sum1 += 1
    # 风热计算
    sum2 = 0 # 命中数
    for i in flu2:
        if checkboxes[i].get():
            ok = 1
        else:
            ok = 0
        if ok == flu2[i]:
            # 命中
            sum2 += 1
    # 命中数 / 症状总数
    if sum1 / len(flu1) >= sum2 / len(flu2):
        result_label.config(text='您大概得了风寒感冒！')
    else:
        result_label.config(text='您大概得了风热感冒！')

root = tk.Tk()
root.title('流感诊断机器人')
# 循环创建选择按钮
symptoms = ['发热', '寒气重', '头痛', '鼻塞', '流黄鼻涕', '流清鼻涕']
checkboxes = {}

for symptom in symptoms:
    checkboxes[symptom] = tk.IntVar()
    tk.Checkbutton(root, text=symptom, variable=checkboxes[symptom]).pack(side='left')

send_button = tk.Button(root, text='诊断', command=send)
send_button.pack(side='left')
# 提示框
result_label = tk.Label(root, text='请勾选您的症状！')
result_label.pack(fill="x")

root.mainloop()